Python list - removing some list entries